import specialized_agents.ui_agent
import json
import uuid

def ui_agent_process_command(
    prompt: str,
    max_step: int = 100,
    max_image: int = 1,
    task_id: str = None,
    requester: str = "RootAgent"
) -> str:
    """向ui-tars-server发送任务请求。

    Args:
        prompt (str): UI任务的提示。
        max_step (int): UI任务的最大步数。默认为100。
        max_image (int): 要生成的最大图像数。默认为1。
        task_id (str, optional): 任务ID。如果为None，将生成一个新的UUID。默认为None。
        requester (str, optional): 任务的请求者。默认为“RootAgent”。

    Returns:
        str: 表示来自UI-TARS服务器响应的JSON字符串。
    """
    if task_id is None:
        task_id = str(uuid.uuid4())

    request_data = {
        "protocol_version": "1.0",
        "task_id": task_id,
        "requester": requester,
        "task": {
            "prompt": prompt,
            "max_step": max_step,
            "max_image": max_image
        }
    }
    
    ui_agent = specialized_agents.ui_agent.UIAgent()
    return ui_agent.process_command(json.dumps(request_data))